# -*- coding: UTF-8 -*-
import torch

import train
import test
import predict
from modle.moduleConfig import Config
from transformers import logging
import logging as outLog

logging.set_verbosity_error()

if __name__ == '__main__':
    outLog.basicConfig(format='[%(asctime)s][*%(levelname)s]:%(message)s',
                       level=logging.INFO)
    # print()
    train_config = Config({
        'epoch': 1,
        'batch_size': 128,
        'pad_size': 35,
        'hidden_size': 768,
        'filter_sizes': (2, 3, 4),
        'num_filters': 256
    })
    # train
    # train.train(config=train_config)
    # test
    # test.test('D:\\myInterestTest\\objectDetect\\NLP\\output\\module_r_20.pth', config=train_config)
    # predict
    # predict.predict('D:\\myInterestTest\\objectDetect\\NLP\\output\\module_r_1.pth', config=train_config)
